package com.hy.study.algorithm.kmp;

import java.util.Arrays;

/**
 * KMP算法匹配字串
 */
public class KMPAlgorithm {
    public static void main(String[] args) {
        String st1 = "BBC ABCDAB ABCDABCDABDE";
        String st2 = "ABCDABD";
        int[] next=kmpNext(st2);
        System.out.println(Arrays.toString(next));
        int i = kmpSearch(st1, st2, next);
        System.out.println(i);
    }

    /**
     * KMP搜索
     * @param str1 原字符串
     * @param str2 字串
     * @param next 字串对应的部分匹配表
     * @return -1 没有匹配到 否则返回第一个匹配的位置
     */
    public  static  int kmpSearch(String str1,String str2,int [] next){
        for (int i = 0,j=0; i < str1.length(); i++) {
            //需要考虑 str1.charAt(i)!=str2.charAt(j)
            while (j>0 && str1.charAt(i)!=str2.charAt(j)){
                j=next[j-1];
            }
            if(str1.charAt(i)==str2.charAt(j)){
                j++;
            }
            if(j==str2.length()){
                //找到了
                return i-j+1;
            }
        }
        return -1;
    }



    /**
     * 获取额一个字符串的部分匹配值 表
     * @param dest 目标字符串
     * @return
     */
    public static int[] kmpNext(String dest) {
        //用来保存部分匹配值
        int[] next = new int[dest.length()];
        //如果字符串的长度1 部分匹配值就是0
        next[0] = 0;
        for (int i = 1, j = 0; i < dest.length(); i++) {
            while (j>0  && dest.charAt(i)!=dest.charAt(j)){
                j=next[j-1];
            }
            //当这个条件满足时 部分匹配值就需要加1
            if (dest.charAt(i) == dest.charAt(j)) {
                j++;
            }
            next[i] = j;
        }
        return next;
    }
}
